#!/bin/bash

# Ollama 模型初始化脚本
# 用于拉取和准备所需的模型

set -e

echo "🚀 初始化 Ollama 模型..."

# 检查 Ollama 是否安装
if ! command -v ollama &> /dev/null; then
    echo "❌ Ollama 未安装，请先安装 Ollama"
    echo "安装方法：https://ollama.com/download"
    exit 1
fi

# 检查 Ollama 服务是否运行
if ! curl -f http://localhost:11434/api/tags &> /dev/null; then
    echo "❌ Ollama 服务未运行，请先启动 Ollama"
    echo "启动方法：ollama serve"
    exit 1
fi

# 拉取配置的模型
MODEL_NAME=${OLLAMA_MODEL_NAME:-"qwen2.5:7b"}

echo "📥 正在拉取模型: $MODEL_NAME"
ollama pull "$MODEL_NAME"

# 验证模型是否可用
echo "🔍 验证模型..."
if ollama list | grep -q "$MODEL_NAME"; then
    echo "✅ 模型 $MODEL_NAME 已成功安装"
else
    echo "❌ 模型 $MODEL_NAME 安装失败"
    exit 1
fi

# 显示已安装的模型
echo "📋 当前已安装的模型："
ollama list

echo "🎉 模型初始化完成！"
